Fixed bug with dtype and cleaned up mnist for PEP8 compliance