Correct comment and add enum for readability