Update Code

Made it easier to read.