Added some small comments and cleaned things up.