add some method in the code