tidied up the final step