use strings instead of integers here, it is more readable in other parts of the code (now you would need to explain in comments that 0 = basic,... everywhere)