Broke it for now but almost there with some icons and menu