Added code for marking deployment complete