Allow users to choose which margin to use

Closes #13