Add testbeam code