Update beam.py
update code completion datasets
move into 4 folders
update inference in line completion
