Wrote unit tests to validate user input