Second commit, added methods and main class with code