Code done, may need fixing

getting low accuracies