The code a bit tidied