Updated TestCodeParser to follow changes in PKB