Adding tests for component adding/removal