simple

improving code