update code for checkstyle