wrote tests, cleaned up code