Done it in the class