WIP: setup, __init__ and unit test