Improved support for export to Brian2