Tidied code up