Add full test coverage