added matplotlib commands in comment