Put screen callback in