playing around with MNSIT dataset