Beautiful lab
Initial state generation enhancements