Added .py extension
Added device selection