Add arguments channelaxis and force_2d

This is a potential solution to #16.