Closes #8. Imported the wrong class XD