missing argument bug fixed
structuring/comments