Added @Enumerated annotation