Add tests to controllers