diff --git a/sample_app.py b/sample_app.py
index 93a9ac1..6142701 100644
--- a/sample_app.py
+++ b/sample_app.py
@@ -5,7 +5,7 @@
 # Sample function created for demonstration of creating custom applications functionality
 # See also https://docs.cyfronet.pl/display/ISDOC/Creating+Custom+Applications
 
-from numpy import loadtxt
+import numpy as np
 import matplotlib.pyplot as plt
 
 def main(numbers_file, multiplier):
@@ -20,8 +20,10 @@ def main(numbers_file, multiplier):
         numbers.
     """
 
-    numbers = loadtxt(numbers_file, comments="#", delimiter=",", unpack=False)
+    numbers = np.genfromtxt(numbers_file, comments="#", delimiter=",", unpack=False)
+    numbers = numbers[~np.isnan(numbers)]
+    numbers = np.array(numbers).flatten()
     multiplied_numbers = numbers * multiplier
     plt.figure()
     plt.plot(multiplied_numbers)
-    plt.savefig('multiplied_numbers_plot.png')
+    plt.savefig('multiplied_numbers_plot.png')
\ No newline at end of file