Use an enum for more elegant code