Update Code

Put cleanup code