Adds more tests and fixes shed.h import