Final version of the code

This is the final version of the code that we had used and showed to
the judges