removed 'if name = main' testing. Testing available in separate file