Still setting up the class functions