Clean code a little bit. Create some enums.