Adding tests for removing images