Cleaning

Remove unused variable and import