Tidy up
Add matrix