changed the code to reflect the requirements of the manually annotated
dataset