add in validation, tests pass for it