The model is getting really low accuracy. This is my first time writing a neural network so I dont really know how to make it better

```
import tensorflow as tf
import matplotlib.pyplot as plt

    #data set
    data = tf.keras.datasets.cifar10

    (x_train, y_train), (x_test, y_test) = data.load_data()
    plt.imshow(x_train[0], cmap=plt.cm.binary)

    #normalize data
    x_train = tf.keras.utils.normalize(x_train, axis=1)
    x_test = tf.keras.utils.normalize(x_test, axis=1)

    #building AI model

    model = tf.keras.models.Sequential()
    model.add(tf.keras.layers.Flatten())
    model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
    model.add(tf.keras.layers.Dense(128, activation=tf.nn.relu))
    model.add(tf.keras.layers.Dense(10, activation=tf.nn.softmax))

    #compile model
    model.compile(optimizer='adam',
                 loss='sparse_categorical_crossentropy',
                 metrics=['accuracy'])

    plt.show()
    #train AI model
    model.fit(x_train, y_train, epochs=3)

```