little edit for assingment