PS fixed
Code tidy
fin