Removed unnecessary comments and refactored a little