added imshow for debug purpose
formatted the code
first step in fixing the rotation of the model