little modifications on the assinments