Few small changes, added ohai gem