refine screen; add py.test