Adding player class -> modified code to stick with it