Python code plots data from file