Small change, trying to get keypress to work