Add some code in class