diff --git a/include/heightmap.hpp b/include/heightmap.hpp
index 4e0e2d43c82b4f6380b9290c4b01c87d5a2e52f5..05442b0f7efb7dd3232f97b50ff3c4b29e9d080c 100644
--- a/include/heightmap.hpp
+++ b/include/heightmap.hpp
@@ -22,7 +22,7 @@ public:
 	bool is(const std::string filename) const;
 
 	//inherited methods
-	virtual void draw(glm::mat4 projection, glm::mat4 view, glm::mat4 model) override;
+	virtual void draw(glm::mat4 projection, glm::mat4 view, glm::mat4 model, glm::vec3 camera_pos) override;
 	virtual void upload() override;
 	virtual void unload() override;
 
@@ -32,6 +32,14 @@ public:
 	void updatePlotWindow();
 
   float minZ() const;
+	float minX() const;
+	float maxX() const;
+	float minY() const;
+	float maxY() const;
+
+	//Shader control
+	void enableLighting();
+	void disableLighting();
 
 private:
 	void _plotHeightLineAlongX(const float y);