Improving code

Restarting part of the code has been lost