Added ability to drop to initial example and made it more sensible