Added module for localization, and modified main.py to use it