Slight cleanup, updated screenshot