Fix tests and updated version number to 0.1.6