remove variable dist
remove print
remove unnecessary files
change input