added more customizable drawing on mnist