Change added to snaiths code