update code

On my way again