Updated for newer keypress module