Add option to change potential type.