added some code in for class to work from