tidy up code
test