added design for UI