Inline If Statement Python