Added key press support