Added python class

For testing.