Played around with classes