Städat kod och lagt in try catch.