tidy up print statements