ENH: Pipeline working. Lost fitting though