Fix failing tests
Add menu layout test
Add tests for menu component