added source i wrote as first homework