Adding plot function

Signed-off-by: Cristian Maureira-Fredes <cmaurei@aei.mpg.de>