Made a workable unit test and implemented for/if keywords