[#1] cleaning code

Just making sure the code looks good. Very nice project!