Wrote some more code :D