Got lost in the if else madness :)