There seems to be a lot of bugs in the code especially the padding and end of sequence indexing. Can you please update the repo with bug free code used to reproduce results in paper. Also, can you share the exact requirements.txt or a Dockerfile. Thanks.