Add code to other classes