Add production and staging checks