Cleans up the code a tiny bit and adds link to source code