WORKING NN, need to clean tho