3rd homework code added