hopefully final corrections for assignment 4