Work on logic, fixed mouse input