Added some arguments instead of hardcoding it.