cleaned

code cleaned up a bit.