Cleaned up code, removed un needed comment