test: run mypc to check types and add annotations. run black and reformatted all python files. run tests and still pass. Closes #20
