Fixes/improves controls. A lot!