great improve
add info class