def units(comp_sum, uncomp_sum):
    """Calculate package size
    """
    compressed = round(sum(map(float, comp_sum)) / 1024, 2)
    uncompressed = round(sum(map(float, uncomp_sum)) / 1024, 2)
    comp_unit = uncomp_unit = 'Mb'
    if compressed > 1024:
        compressed = round(compressed / 1024, 2)
        comp_unit = 'Gb'
    if uncompressed > 1024:
        uncompressed = round(uncompressed / 1024, 2)
        uncomp_unit = 'Gb'
    if compressed < 1:
        compressed = sum(map(int, comp_sum))
        comp_unit = 'Kb'
    if uncompressed < 1:
        uncompressed = sum(map(int, uncomp_sum))
        uncomp_unit = 'Kb'
    return [comp_unit, uncomp_unit], [compressed, uncompressed]
