Better IDE integration