updated code forgotten on last commit