Add tests for status code