added stuff to Assignment-1