updates to the code