Code to practice python and CS basics