Part of the code Cc:@dehmey