Added transfer_matrix and potential initializations to main.cpp