Tidy up plus Fix