Started rewrite on plotting in python