Fixed identation and added comments