Adding tests (all pass)