tidying up code :pizza::cat: