TODO about fixing reading of the input data