Increasing test coverage and fixing bugs found while writing tests

`y` is just a 1-dimensional array though so I just want to increase its length so that it doesn't match the number of instances in the data (i.e., the number of feature rows).