E ai cara.
Então, onde tiver um if similar a este (
if (e.which == 40)), basta trocar o número pelo número correspondente a tecla que deseja.
Nesse link(
https://api.jquery.com/keypress/), lá no final da descrição, tem um exemplo, onde você pressiona a tecla que quer e ele todas as infos da tecla pressionada.