Tidied the code formatting