Should consider using an enum here