Cleaned up the code little