Got tests written :)