Tidy up test code
Added the installation section