simplify function.
Add __init__ which was left out ?