added starter code for UI