updtes to the code