removed weird code
4 tests work