A separate Stages class now contains the code for individual stages, making for a cleaner Squared class.