Added covariance type argument, where diag is added as an option, fixes #32