updated the code structure