Code at the end of class