init: Fix for handling -p argument in the write command

We now correctly process the -p argument even if it is not the last
argument passed in to the function. The earlier iteration used to
incorrectly skip over it under some cases.

Change-Id: I842ce6f62a85b19a79bef4c082e132d1c9649485