changed main to allow for a user input as class size.