Minor change to get info from UI