Handling keyboard control. Some visual improves.