Added testing files and changed get_distribution structure