style(segmentation): PEP8
feat(segmentation): merge PK and WindowDiff

@guinaudeau can you please check that the changes I made are correct?