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