Minor fixes to comments, adding variable length inputs to solve