Tidy up input loop