Adding initial plotting code.