Everything is pretty much working. Still need to make the output pretty