Cleaned and re-formatted up the code slightly