Fixing my display py