Imported from https://github.com/MatanRad/Neural-IR-Project