changed the input a bit