Fixed up keyboard controls