Refactor and add comment