[#4] Structure the code as a python package