updates on the code