added variable to cleanup code