diff --git a/ska_python_skeleton/main.py b/ska_python_skeleton/ska_python_skeleton.py
similarity index 85%
rename from ska_python_skeleton/main.py
rename to ska_python_skeleton/ska_python_skeleton.py
index 3a106ce8cb42c58f6b29d8680e4c884a7d075008..7dd613e2c0084d4a358082667939c3fb70b994b7 100644
--- a/ska_python_skeleton/main.py
+++ b/ska_python_skeleton/ska_python_skeleton.py
@@ -9,12 +9,14 @@ def function_example():
 
 
 # TODO: Replace all the following code with the desired functionality for the package
-class SKA():
-    """Define class and it's attributes"""
+class SKA:
+    """Define class, methods etc"""
+
+    something = 0
 
     def example(self):
         """Example: Define non return function for subsequent test."""
 
-    def testing_example(self):
+    def example_2(self):
         """Example: Define function for subsequent test with specific return value."""
         return 2
diff --git a/tests/test_ska_skeleton.py b/tests/test_ska_skeleton.py
index 77b92f949d3671d60a0c18ef709c84848487718f..553d1790b64aa674ab19afa33393738b8729d40e 100644
--- a/tests/test_ska_skeleton.py
+++ b/tests/test_ska_skeleton.py
@@ -4,7 +4,7 @@
 """Tests for the ska_python_skeleton module."""
 import pytest
 
-from ska_python_skeleton import main
+from ska_python_skeleton import ska_python_skeleton
 
 
 # TODO: Replace all the following examples with tests for the ska_python_skeleton package code
@@ -34,5 +34,7 @@ def test_ska_python_skeleton(an_object):
 
 def test_package():
     """Example: Assert the ska_python_skeleton package code."""
-    assert main.example() is None
-    assert main.testing_example() == 2
+    assert ska_python_skeleton.function_example() is None
+    foo = ska_python_skeleton.SKA()
+    assert foo.example_2() == 2
+    assert foo.example() is None