Maybe a enum would be better than string? But I'm ok with string as well