#ifndef UNIT_BUTTON_H
#define UNIT_BUTTON_H

#include <button.h>

namespace battleship{
	class Unit;

	class UnitButton : public vb01Gui::Button{
		public:
			UnitButton(vb01::Vector3, vb01::Vector2, std::string, std::string, int, std::string);
		protected:
			std::vector<Unit*> getUnits();
	};
}

#endif