Trying plotting in python