adding in some slight modifications to enable us to start writing test