stoikov: Tidy up plotting script

Signed-off-by: Aleksi Pitkäjärvi <aleksi.pitkajarvi@aalto.fi>