fix bugs
add missing files