Updates to the Code