Added code to make it compile with gradle