Matplotlib plot () function in python