Adding assignament code