Added proper inheritance and buttons unit tests.