would it be ok to add the following parameter?
get_free_space(directory, size = None)
format_size(num, size)

This way, get_free_space will be able to return space in the specified size or return the size like it normally would when size not specified.