add new code to day 4