I was just looking at one more thing. To improve the accessibility of sub menus, this should be added based on if the sub menu is open or closed.
aria-expanded (false) aria-expanded (true)
Any suggestions on how I can incorporate this in to my existing code? Sorry for coming back like this, just trying to design without mistakes.