Update __init__.py

self.mode should default to 'TEST' if I am in sandbox. otherwise go back to default 'LIVE' mode. Unless I am getting something wrong here, I don't see the need to have a method to set_mode ? Hope this is useful!