Added labels back into the UI