Using visual studio code for python