Added collapse UI feature