Improve bedrock gen