Created the fkp_class and updated the notebook to use it. Works fine