Add comments and change readme