Merged training repo into here
Refactored beam on sign hack