removed dependency to PyYAML and removed unused imports