Remove default imports
Add stages schema plot