End of class

Almost complete function