A couple a quick changes