Add more tests to component