I need to rewrite the function I guess.