Wrote more for extra credit task