Mapped out battle a little more and created Button class with a light UI