(WIP) Still doesn't work on first keypress, everything else works.