Code cleanup

Removed some extra lines