Changed structure of input layer.
Changed the Data structure of the input layer, modified the output slightly.