Update mui code. This is the code found in Habitrac App