Now, basic example is working properly. Need to create a class however.