Updated src file to reflect changes