cleaned up code a little bit.
cleaned up code a little bit.