Added Tests
Deleted unnecessary line of code