updated mlp test to reflect layer arg shape changes
recent changes
added mnist plotting