tidied up code
still the same