Added: Ehsan solution for the first assignment