Added a function, might change later.