Added Button class to solution