Fix indexing with `Bool`

Fix bool indexing (#27)

* Fix #26

* Also fix boolean indexing for Ones/Zeros

* Rewrite to make sure that sizes are checked correctly for non-vectors.

* Add tests to check for dimension mismatch

* Add missing tests for bool indexing