Added conclusion & edited code