Added helpful assertion
Filling out tests