Removed 2 extra code lines