Ask for input more prettily
Might as well use a module