Added more functionality to player_class.rb script