Adds tests for new methods