Add unit tests for user input