fix errors
add & fix codes