fixed up input a bit