Assignment for class… Don’t like the end part though