Adding in cases to cover additional cases