Updated to latest mui components