Test code for armadillo - still works :)