Update clean up

Clean up