Small updates and update the readme