Added initial implementation of mnist