updating Documentation

cleaned code a little