remove unused import
Add session lookup