[FEATURE] Fixed checkstyle errors