Shorter implementation of p21