Added initial plotting code