working version - check the wasp next