Update UI to work with improved Rails API