Adding tests for different inputs