added my solution for first assignment