Updated to work with LB API