Add missing code ending