Updated ui to work with updated api