Finished up all the code, need to debug and perform experiments