Training and Testing our Model