Nearly working, just need to change the output.