refactor/ comment

Made things slightly clearer