Fixed plot commands & removed unused imports