small corrections for assignment