Adds tests for remove attribute