add files and correct bug