I feel like this is kinda weird, using java classes as our type system. See potential solution of using an enum below: