More documentation and update readme