little tweaks between the functions