Add comments and refactor