Done algorithm.py prototyping