Add unit tests for invalid inputs