Revamped the design UI