Computing p-canonical bases