Add '__init' and '__exit'