Remove return outside of function