Doing some modification in class - Code Works