erm, could you write required code modifications please?