Fix imports
Add logic to answering challenges