Adding __init__

Signed-off-by: Cristian Maureira-Fredes <cmaurei@aei.mpg.de>