added simple code to test how classes work in python