Added bits of ui