fixed styling and added 2 tests