Code works

Found the bug, there was a messed up file in one of the data file that
was breaking the code. Fixed it.