Cleanup extra import
fix screen