Added some tests to the component