Add tests for income