Added @Before and @After. Pretty code <3