simplify code
tidy up
tidy up