added another little bit to the code