Started to working on fixing the test after new entities have been created. Tried some stuff in the main class to get the idea of the sequence.