- Fixed test cases
- Fixed return statement again