Change n-body visualiser to use new API