Made the solution run, added input handling