Basic modle done just need to finish and clean up the code and start
testing