Removed extra import from lwap function in queries.py - plot behaviour has changed?