UI improved with new graphics