Cleaned up the code again

You did alot of document.getElementById, but you can also do a reference to a element in a variable. Which make it much easier to read.