finished coursework, need to write main and test