Added rough code for locate class

Still needs to initialize data and make sense of output and add trained
weight values