Fixed imports, updated component to use the library