added solution to first assignment