Fix + some clean-up...
...