Fixed import issues and prepared the setup.py
Fixed an issue where visualization types were not correctly read from the configs.