playing around with swift making a class