Let the "write-tests" prompt ask to extend the given source file in c…

Let the "write-tests" prompt ask to extend the given source file in c… #1046