Testing out python classes