Added Mock UI class