Hi, I updated the code to reflect, sorry its been a while since i forked the repo so forgot the reasoning for this!